import rasterio
import matplotlib.pyplot as plt

input_path = '../../data/picture/output_tif/train/label/patch_0_0.tif'

with rasterio.open(input_path) as src:
    # 获取影像的宽度和高度
    width = src.width
    height = src.height
    print(width, height)